bzoj3172 [Tjoi2013]单词

发表于 
1
2
3
4
5
fail指针表示最长后缀
每个单词的每个位置对它的所有后缀都有贡献
也就是每个单词的答案为fail树中这个点的子树和
注意一个单词的每个位置都有1的初值
可以不用把fail树单独建出来，处理bfs序就行
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#include<bits/stdc++.h>
#define mp make_pair
#define pb push_back
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
inline LL read()
{
	LL x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
	return x*f;
}
const int N=1000008;
int n;
char s[N];
struct node{
	int ch[26],fail,num;
}t[N];
int sz;
int pos[208];
inline void inser(int &x)
{
	int u=0,n=strlen(s+1),c;
	for(int i=1;i<=n;++i){
		c=s[i]-'a';
		if(!t[u].ch[c])	t[u].ch[c]=++sz;
		u=t[u].ch[c];
		++t[u].num;
	}
	x=u;
}
int q[N],he,ta;
inline void getAC()
{
	he=1;ta=1;
	for(int i=0;i<26;++i){
		if(t[0].ch[i]){
			q[ta]=t[0].ch[i];
			++ta;
		}
	}
	int u,v;
	while(he!=ta){
		u=q[he];
		++he;
		for(int i=0;i<26;++i){
			v=t[u].ch[i];
			if(!v){
				t[u].ch[i]=t[t[u].fail].ch[i];
				continue;
			}
			t[v].fail=t[t[u].fail].ch[i];
			q[ta]=v;
			++ta;
		}
	}
	for(int i=ta-1;i;--i){
		t[t[q[i]].fail].num+=t[q[i]].num;
	}
}
int main()
{
	n=read();
	for(int i=1;i<=n;++i){
		scanf("%s",s+1);
		inser(pos[i]);	
	}
	getAC();
	for(int i=1;i<=n;++i){
		printf("%d\n",t[pos[i]].num);
	}
	return 0;
}